from ultralytics import YOLO




if __name__ == '__main__':


    # Load a model
    model = YOLO("yolo11s-pose.yaml")  # build a new model from YAML
    model = YOLO("yolo11s-pose.pt")  # load a pretrained model (recommended for training)
    model = YOLO("yolo11s-pose.yaml").load("yolo11s-pose.pt")  # build from YAML and transfer weights

    # Train the model
    results = model.train(data="../conf/petpose.yaml", epochs=10, imgsz=640,batch=24,mixup=0.5)